package com.sheng.leetcode.year2023.month06.day29;

import org.junit.Test;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.stream.Collectors;

/**
 * @author liusheng
 * @date 2023/06/29
 * <p>
 * 1253. 重构 2 行二进制矩阵<p>
 * <p>
 * 给你一个 2 行 n 列的二进制数组：<p>
 * 矩阵是一个二进制矩阵，这意味着矩阵中的每个元素不是 0 就是 1。<p>
 * 第 0 行的元素之和为 upper。<p>
 * 第 1 行的元素之和为 lower。<p>
 * 第 i 列（从 0 开始编号）的元素之和为 colsum[i]，colsum 是一个长度为 n 的整数数组。<p>
 * 你需要利用 upper，lower 和 colsum 来重构这个矩阵，并以二维整数数组的形式返回它。<p>
 * 如果有多个不同的答案，那么任意一个都可以通过本题。<p>
 * 如果不存在符合要求的答案，就请返回一个空的二维数组。<p>
 * <p>
 * 示例 1：<p>
 * 输入：upper = 2, lower = 1, colsum = [1,1,1]<p>
 * 输出：[[1,1,0],[0,0,1]]<p>
 * 解释：[[1,0,1],[0,1,0]] 和 [[0,1,1],[1,0,0]] 也是正确答案。<p>
 * <p>
 * 示例 2：<p>
 * 输入：upper = 2, lower = 3, colsum = [2,2,1,1]<p>
 * 输出：[]<p>
 * <p>
 * 示例 3：<p>
 * 输入：upper = 5, lower = 5, colsum = [2,1,2,0,1,0,1,2,0,1]<p>
 * 输出：[[1,1,1,0,1,0,0,1,0,0],[1,0,1,0,0,0,1,1,0,1]]<p>
 * <p>
 * 提示：<p>
 * 1 <= colsum.length <= 10^5<p>
 * 0 <= upper, lower <= colsum.length<p>
 * 0 <= colsum[i] <= 2<p>
 * <p>
 * 来源：力扣（LeetCode）<p>
 * 链接：<a href="https://leetcode.cn/problems/reconstruct-a-2-row-binary-matrix">1253. 重构 2 行二进制矩阵</a><p>
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。<p>
 */
public class LeetCode1253 {

    @Test
    public void test01() {
//        int upper = 2, lower = 1;
//        int[] colsum = {1, 1, 1};
        int upper = 2, lower = 3;
        int[] colsum = {2, 2, 1, 1};
//        int upper = 5, lower = 5;
//        int[] colsum = {2, 1, 2, 0, 1, 0, 1, 2, 0, 1};
        System.out.println(new Solution().reconstructMatrix(upper, lower, colsum));
    }
}

class Solution {
    public List<List<Integer>> reconstructMatrix(int upper, int lower, int[] colsum) {
        // upper 为第零行元素之和，lower 为第一行元素之和
        int n = colsum.length;
        int[][] nums = new int[2][n];
        for (int i = 0; i < n; i++) {
            if (colsum[i] == 0) {
                nums[0][i] = 0;
                nums[1][i] = 0;
            } else if (colsum[i] == 2) {
                nums[0][i] = 1;
                nums[1][i] = 1;
                upper--;
                lower--;
            } else {
                if (upper >= lower) {
                    nums[0][i] = 1;
                    nums[1][i] = 0;
                    upper--;
                } else {
                    nums[1][i] = 1;
                    nums[0][i] = 0;
                    lower--;
                }
            }
        }
        return upper == 0 && upper == lower ? Arrays.stream(nums).map(i -> Arrays.stream(i).boxed().collect(Collectors.toList())).collect(Collectors.toList()) : new ArrayList<>();
    }
}
